#include <stdio.h>
// atoi函数
#include <stdlib.h>
// I/O函数
#include <unistd.h>
#include <time.h>
// socket 系列函数
#include <sys/socket.h>
// sockaddr_in 结构体
#include <netinet/in.h>
// hton 函数
#include <arpa/inet.h>
#include <signal.h>
// memset函数
#include <string.h>
// 服务器socket描述符
int server_fd = 0;
// 服务器连接属性
struct sockaddr_in server_attr;
// 客户端socket描述符
int client_fd = 0;
// 客户端连接属性
struct sockaddr_in client_attr;
// ctrl + c 关闭服务器监听
void sigint_handle(int);
// 响应客户端连接
void response(int);
// 服务器连接日志
void server_log(struct sockaddr_in*);
int main(int argc,char* argv[])
{
	if(argc < 2)
	{
		printf("incorrect parameter\n");
		return 1;
	}
	if(SIG_ERR == signal(SIGINT,sigint_handle))
	{
		perror("signal error");
		return 1;
	}
	memset(&server_attr,0,sizeof(server_attr));
	memset(&client_attr,0,sizeof(server_attr));
	/*
 	 * 1. 创建socket
	 * socket创建在内核中，是一个结构体
	 * AF_INET : IPV4
	 * SOCK_STREAM : TCP协议
	 */
	server_fd = socket(AF_INET,SOCK_STREAM,0);
	if(server_fd == -1)
	{
		perror("sock error");
		return 1;
	}
	/*
 	 * 2. 调用bind函数将socket与地址(ip,port)进行绑定
 	 */
	server_attr.sin_family = AF_INET;
	// 转换位网络字节序
	server_attr.sin_port = htons((short)atoi(argv[1]));
	// 表示所有ip，也可以指定第一个ip，指定ip时要把主机字节序转换成网络字节序
	server_attr.sin_addr.s_addr = INADDR_ANY;
	// 绑定ip端口
	if(-1 == bind(server_fd,(struct sockaddr*)&server_attr,sizeof(server_attr)))
	{
		perror("bind error");
		return 1;
	}
	/*
 	 * 3. 调用listen函数启动监听，通知系统接收来自客户端的请求
 	 * 第二个参数代表客户端队列的长度
 	 * 执行成功后能用netstat查看
 	 */
	if(-1 == listen(server_fd,8))
	{
		perror("listen error");
		return 1;
	}
	socklen_t addrlen = sizeof(client_attr);
	/*
	 * 4. 调用accept获得客户端连接，并返回一个新的socket文件描述符
	 * 若没有客户端连接，此函数会阻塞，直到获得一个客户端连接
  	 */
	while(1)
	{
		client_fd = accept(server_fd,(struct sockaddr*)&client_attr,&addrlen);
		server_log(&client_attr);
		/*
 		 * 5. 调用I/O函数(read/write)和连接的客户端进行双向的通信
 		 */
		response(client_fd);
		close(client_fd);
	}
	close(server_fd);
}
void sigint_handle(int sig)
{
	close(server_fd);
	exit(0);
}
void response(int fp)
{
	time_t rawtime;
	struct tm* timeinfo;
	time(&rawtime);
	timeinfo = localtime(&rawtime);
	char str[39] = "I am server. ";
	strcat(str,asctime(timeinfo));
	write(fp,str,sizeof(str));
}
void server_log(struct sockaddr_in* client_attr)
{
	char ip[16] = {'\0'};
	// 结构体的的数据不能直接显示，需要先转换成本机字节序
	inet_ntop(AF_INET,&client_attr->sin_addr.s_addr,ip,sizeof(ip));
	printf("connected by %s:%d\n",ip,ntohs(client_attr->sin_port));
}
